package com.nowcoder.community.service;

import com.nowcoder.community.util.RedisKeyUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.core.RedisOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.SessionCallback;
import org.springframework.stereotype.Service;


/**
 * 点赞功能
 *      支持对帖子 评论 用户 点赞
 *      第一次 点赞  再来则  取消
 * 首页点赞数量
 *      统计点赞数量
 *
 */
@Service
public class LikeService {

    @Autowired
    private RedisTemplate redisTemplate;

    // 点赞
    public void like(int userId, int entityType, int entityId, int entityUserId) {
        redisTemplate.execute(new SessionCallback() {
            @Override
            public Object execute(RedisOperations operations) throws DataAccessException {
                String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
                String userLikeKey = RedisKeyUtil.getUserLikeKey(entityUserId);

                // 判断这个用户userId有没有对这个实体entity（可以是帖子、评论、 用户）点赞 (其实可以使用下方函数)
                boolean isMember = operations.opsForSet().isMember(entityLikeKey, userId);
                // 开启事务
                operations.multi();

                // 注意mysql中自增、自减是原子性操作，因此我们这里不用当心高并发下点赞数量出错问题
                if (isMember) {
                    // 如果已经点过赞了，那么把userId从实体获得的点赞集合中移除掉
                    operations.opsForSet().remove(entityLikeKey, userId);
                    // 如果已经点过赞了，那么把userId从实体对应用户获得的点赞集合中减一
                    operations.opsForValue().decrement(userLikeKey);
                } else {
                    operations.opsForSet().add(entityLikeKey, userId);
                    operations.opsForValue().increment(userLikeKey);
                }

                return operations.exec();
            }
        });
    }

    /**
     * 查询某实体点赞的数量
     * @param entityType 实体类型
     * @param entityId 实体id
     * @return 实体获得的赞的数量 其实就是Redis Set的统计数量方法
     */
    public long findEntityLikeCount(int entityType, int entityId) {
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        return redisTemplate.opsForSet().size(entityLikeKey);
    }

    /**
     * 查询某人对某实体的点赞状态
     * @param userId 用户id
     * @param entityType 实体类型
     * @param entityId 实体id
     * @return 整数，表示状态，方便后期扩展；由于boolean值只有两种情况，要是后期除了点赞还有点踩等多种状态时，int值表示多种情况
     *          实现方法：本质是利用Set的判断是否存在该元素的方法
     */
    public int findEntityLikeStatus(int userId, int entityType, int entityId) {
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        return redisTemplate.opsForSet().isMember(entityLikeKey, userId) ? 1 : 0;
    }

    /**
     * 查询某个用户获得的赞
     * @param userId 用户id
     * @return 用户获得的赞数量
     */
    public int findUserLikeCount(int userId) {
        String userLikeKey = RedisKeyUtil.getUserLikeKey(userId);
        Integer count = (Integer) redisTemplate.opsForValue().get(userLikeKey);
        return count == null ? 0 : count.intValue();
    }

}
